codebase_config = dict(type='mmseg', task='Segmentation', with_argmax=True)
backend_config = dict(type='onnxruntime')
onnx_config = dict(
    type='onnx',
    export_params=True,
    keep_initializers_as_inputs=False,
    opset_version=11,
    save_file='pidnet-s_2xb6-120k_512x512-roadcracks.onnx',
    input_names=['input'],
    output_names=['output'],
    optimize=True,
    input_shape=[512, 512])



